#pragma once
#ifndef CBBOX_FISH_H_
#define CBBOX_FISH_H_

#include <cstdint>
#include <pybind11/pybind11.h>

class cBBox_fish
{
private:
    const int batch_size;
public:
    cBBox_fish(int bs, float vt);
    ~cBBox_fish();
    void py_compute(pybind11::buffer ibuf, pybind11::buffer obuf);
    void py_compute_fish(pybind11::buffer ibuf, pybind11::buffer obuf);
    void compute(const int16_t input[][2][7], int32_t result[][4]);
    void compute_fish(const int16_t input[][2][7], int32_t result[][4]);
};

#endif
